from selenium import webdriver
import time


def rest(sec):
    time.sleep(sec)


if __name__ == '__main__':
    browser = webdriver.Chrome()
    url = 'https://www.baidu.com'
    url2 = 'https://www.taobao.com'
    url_python = 'https://python.org'
    try:
        browser.get(url)
        # js 的 window.open() 方式打开新标签页
        browser.execute_script('window.open()')
        # 打印所有 window 的 handles
        print(browser.window_handles)
        last_handle = browser.current_window_handle
        print("当前的 handle 是", browser.current_window_handle)
        # 切换到第二个 window, 并访问淘宝
        browser.switch_to.window(browser.window_handles[1])
        browser.get(url2)
        # 切换回原先的百度 window, 然后去 python 网站
        rest(1)
        browser.switch_to.window(last_handle)
        browser.get(url_python)
    finally:
        rest(5)
        browser.close()
